import torch
import torch.backends.cudnn as cudnn

print(f"PyTorch Version: {torch.__version__}")
print(f"CUDA Available: {torch.cuda.is_available()}")
print(f"CUDNN Enabled: {cudnn.enabled}")
print(f"CUDNN Version: {cudnn.version()}")
print(f"Number of GPUs: {torch.cuda.device_count()}")
print(f"Current GPU: {torch.cuda.current_device()}")
print(f"GPU Name: {torch.cuda.get_device_name(0)}")
